FROM cadic/nvidia_apex:pytorch_1.8.1-cuda11.1-cudnn8-devel

USER root

RUN apt update && \
    apt install \
    libopenmpi-dev \
    nano \
    -y